package entity

import (
	"github.com/volcengine/volcengine-go-sdk/service/arkruntime"
	"github.com/volcengine/volcengine-go-sdk/service/arkruntime/model"
	"os"
)

type ChatClient struct {
	Client       *arkruntime.Client
	BotID        string
	SystemPrompt string
}

func NewChatClient() *ChatClient {
	return &ChatClient{
		Client:       arkruntime.NewClientWithApiKey(os.Getenv("ARK_API_KEY")),
		BotID:        "bot-20250425164816-2p259",
		SystemPrompt: "你是豆包，字节跳动开发的 AI 人工智能助手",
	}
}

type ChatRequest struct {
	Messages []*model.ChatCompletionMessage `json:"messages"`
}

type ChatResponse struct {
	Content    string   `json:"content"`
	References []string `json:"references,omitempty"`
}
